1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
import { createContextFromRequest } from "@/server/api/client";
import { and, eq } from "drizzle-orm";
import { assets } from "@karakeep/db/schema";
import {
createAssetReadStream,
getAssetSize,
readAssetMetadata,
} from "@karakeep/shared/assetdb";
export const dynamic = "force-dynamic";
export async function GET(
request: Request,
{ params }: { params: { assetId: string } },
) {
const ctx = await createContextFromRequest(request);
if (!ctx.user) {
return Response.json({ error: "Unauthorized" }, { status: 401 });
}
const assetDb = await ctx.db.query.assets.findFirst({
where: and(eq(assets.id, params.assetId), eq(assets.userId, ctx.user.id)),
});
if (!assetDb) {
return Response.json({ error: "Asset not found" }, { status: 404 });
}
const [metadata, size] = await Promise.all([
readAssetMetadata({
userId: ctx.user.id,
assetId: params.assetId,
}),
getAssetSize({
userId: ctx.user.id,
assetId: params.assetId,
}),
]);
const range = request.headers.get("Range");
if (range) {
const parts = range.replace(/bytes=/, "").split("-");
const start = parseInt(parts[0], 10);
const end = parts[1] ? parseInt(parts[1], 10) : size - 1;
const stream = createAssetReadStream({
userId: ctx.user.id,
assetId: params.assetId,
start,
end,
});
return new Response(
// eslint-disable-next-line @typescript-eslint/no-unsafe-argument, @typescript-eslint/no-explicit-any
stream as any,
{
status: 206, // Partial Content
headers: {
"Content-Range": `bytes ${start}-${end}/${size}`,
"Accept-Ranges": "bytes",
"Content-Length": (end - start + 1).toString(),
"Content-type": metadata.contentType,
},
},
);
} else {
const stream = createAssetReadStream({
userId: ctx.user.id,
assetId: params.assetId,
});
return new Response(
// eslint-disable-next-line @typescript-eslint/no-unsafe-argument, @typescript-eslint/no-explicit-any
stream as any,
{
status: 200,
headers: {
"Content-Length": size.toString(),
"Content-type": metadata.contentType,
},
},
);
}
}
|
